package com.chilly.vidio.study.leetcode;

import java.util.ArrayList;
import java.util.List;

/**
 * 438. 找到字符串中所有字母异位词
 * 给定一个字符串 s 和一个非空字符串 p，找到 s 中所有是 p 的字母异位词的子串，返回这些子串的起始索引。
 * <p>
 * 字符串只包含小写英文字母，并且字符串 s 和 p 的长度都不超过 20100。
 * <p>
 * 说明：
 * 字母异位词指字母相同，但排列不同的字符串。
 * 不考虑答案输出的顺序。
 * <p>
 * 输入:
 * s: "cbaebabacd" p: "abc"
 * <p>
 * 输出:
 * [0, 6]
 * <p>
 * 解释:
 * 起始索引等于 0 的子串是 "cba", 它是 "abc" 的字母异位词。
 * 起始索引等于 6 的子串是 "bac", 它是 "abc" 的字母异位词。
 *
 * @author Chilly Cui
 * @createDate 2021/3/4
 */
public class No438FindAllAnagramsInString {

    public static void main(String[] args) {
        String s = "cbaebabacd";
        String p = "abc";
        System.out.println(new Solution().findAnagrams(s, p));
    }

    static class Solution {

        //滑动窗口
        public List<Integer> findAnagrams(String s, String p) {
            List<Integer> resultList = new ArrayList<>();

            //字符串只包含小写英文字母
            int[] pFreq = new int[26];
            for (int i = 0; i < p.length(); i++) {
                pFreq[p.charAt(i) - 'a']++;
            }

            int start = 0;
            int end = -1;

            while (start < s.length()) {
                if (end + 1 < s.length() && (end - start + 1) < p.length()) {
                    end++;
                } else {
                    start++;
                }

                // 当窗口内字符串长度等于字符串p的长度时，则判断其是不是字符串p的字母异位词子串
                if (end - start + 1 == p.length() && isAnagrams(s.substring(start, end + 1), pFreq)) {
                    resultList.add(start);
                }
            }
            return resultList;
        }

        // 判断当前子串是不是字符串p的字母异位词
        private boolean isAnagrams(String window, int[] pFreq) {
            // 计算窗口内字符串各元素的出现次数
            int[] windowFreq = new int[26];
            for (int i = 0; i < window.length(); i++) {
                windowFreq[window.charAt(i) - 'a']++;
            }

            // 比较窗口内各元素的出现次数和字符串p中各元素的出现次数是否一样
            // 如果都一样，则说明窗口内的字符串是字符串p的字母异位词子串
            // 如果不一样，则说明不是其子串
            for (int j = 0; j < 26; j++) {
                if (windowFreq[j] != pFreq[j]) {
                    return false;
                }
            }
            return true;
        }
    }

}
